package com.hmdp.service.impl;

import cn.hutool.core.util.BooleanUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;
import com.hmdp.dto.Result;
import com.hmdp.entity.Shop;
import com.hmdp.mapper.ShopMapper;
import com.hmdp.service.IShopService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Service;

import java.util.concurrent.TimeUnit;

import static com.hmdp.constant.RedisConstants.*;

@Service
public class ShopServiceImpl extends ServiceImpl<ShopMapper, Shop> implements IShopService {

    @Autowired
    private StringRedisTemplate stringRedisTemplate;

    /**
     * 根据id查询商铺信息
     *
     * @param id
     * @return
     */
    @Override
    public Result queryById(Long id) {
        // 1.从 Redis 中查询商铺缓存
        String shopJson = stringRedisTemplate.opsForValue().get(CACHE_SHOP_KEY + id);
        if (StrUtil.isNotBlank(shopJson)) {
            // 2.缓存命中，直接返回商铺信息
            Shop shop = JSONUtil.toBean(shopJson, Shop.class);
            return Result.ok(shop);
        }
        // 6.判断是否命中的是空值，防止缓存穿透
        if (shopJson != null) {
            // 7.命中的是空值，直接返回商铺不存在的信息
            return Result.fail("商铺不存在");
        }
        // 8.缓存未命中，尝试获取分布式锁，防止缓存击穿
        Shop shop = null;
        try {
            boolean isLock = tryLock(LOCK_SHOP_KEY + id);
            if (!isLock) {
                // 9.获取锁失败，休眠并重试
                Thread.sleep(50);
                // ！！！递归重试查询商铺信息
                return queryById(id);
            }
            // 10.获取锁成功，查询数据库
            // 3.缓存未命中，查询数据库
            shop = getById(id);
            if (shop == null) {
                // 4.数据库中不存在该商铺，缓存空值到 Redis 中，防止缓存穿透
                stringRedisTemplate.opsForValue().set(CACHE_SHOP_KEY + id, "", CACHE_NULL_TTL, TimeUnit.MINUTES);
                return Result.fail("商铺不存在");
            }
            // 5.数据库中存在该商铺，写入 Redis 缓存
            stringRedisTemplate.opsForValue().set(CACHE_SHOP_KEY + id, JSONUtil.toJsonStr(shop), CACHE_SHOP_TTL, TimeUnit.MINUTES);
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        } finally {
            // 11.释放分布式锁
            unlock(LOCK_SHOP_KEY + id);
        }
        return Result.ok(shop);
    }

    /**
     * 尝试获取分布式锁
     *
     * @param key
     * @return
     */
    private boolean tryLock(String key) {
        Boolean flag = stringRedisTemplate.opsForValue().setIfAbsent(key, "1", 10, TimeUnit.SECONDS);
        return BooleanUtil.isTrue(flag);
    }

    /**
     * 释放分布式锁
     *
     * @param key
     */
    private void unlock(String key) {
        stringRedisTemplate.delete(key);
    }

    /**
     * 更新商铺信息
     *
     * @param shop
     * @return
     */
    @Override
    public Result updateShop(Shop shop) {
        // 1.更新数据库中的商铺信息
        updateById(shop);
        // 2.删除 Redis 中的商铺缓存
        stringRedisTemplate.delete(CACHE_SHOP_KEY + shop.getId());
        // 3.返回更新成功的信息
        return Result.ok();
    }
}
